package gorm2

import (
	"gorm.io/gorm"
	"sync"
)

type Gorm2ShardingDb struct {
	lock      *sync.Mutex
	defaultDb string              // 默认db,没有分表分库策略的走默认
	dbMap     map[string]*gorm.DB // 映射
}

func (db *Gorm2ShardingDb) GetDefaultDb() *gorm.DB {
	return db.dbMap[db.defaultDb]
}

// GetDb 解析sql路由到对应的库
func (db *Gorm2ShardingDb) GetDb(sql string, args ...any) *gorm.DB {
	return db.dbMap[db.defaultDb]
}
